# -*- coding:utf-8 -*-

# @Time    : 2024/04/09
# @Author  : qichen


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# import sys
import asyncio
# sys.path.append('../../../')
from bot.insurance_planner_gpt.agent.base_agent import LLMAgent

default_template_v1 = """
请记住，你是一名专业的{role_name}，你正通过电话和用户沟通，你需要根据你们的对话记录，判断接下来是否需要挂机，
- 判断需要挂机，输出“是”
- 判断不需要挂机，输出“否”

注意以下事项：
1.你只能输出"是"或"否"，不能输出其他内容。
2.主要参考对话记录里{role_name}说的最后一段回复，如果这里说的意思是需要挂机，那么你就输出“是”，比如对话里说了再见、拜拜等都说明需要挂机。

===
{conversation_history}
===

请判断是否需要挂机：

"""


class HangUp(LLMAgent):
    """话术生成-agent"""

    def __init__(self, role_name, conversation_history):
        conversation_history = "会话历史：\n" + conversation_history
        self.prompt = default_template_v1.format(role_name=role_name,
                                                 conversation_history=conversation_history)
        super().__init__(self.prompt)


if __name__ == '__main__':
    conversation_history = """
    保险顾问:您好！我是慧择保险的顾问，非常感谢您一直以来对我们的信任和支持。我们现在为老客户提供了更加个性化的专属服务，可以通过企业微信方式，随时解答您关于保险的疑问，为您提供更贴心的服务体验。这样方便您随时随地都能得到及时的帮助。您看这样可以吗？我们加一下微信，这样日后沟通会更便捷呢。
用户:你谁啊
保险顾问:我是慧择保险经纪的服务人员，慧择是一家专业的保险经纪平台，相当于保险超市，帮用户精挑细选最适合性价比最高的产品。有了我们的专属1对1服务，关于保险的任何问题都可以随时找我聊，非常方便哦。那我这边用官方企业微信添加您，希望您那边操作同意一下。
用户:慧择是啥
保险顾问:慧择保险网是国内领先的在线保险服务平台，提供的是一站式的保险对比购买服务，帮您在众多保险产品中挑选到最合适自己的保障方案。您加我微信后，有任何保险问题都能第一时间得到专业解答，还有不定期的优惠活动通知哦。这样您随时有问题，直接微信找我，非常方便。
用户:哦
保险顾问:喂，您好（空1s）诶，你好，不好意思再次打扰您了，我是慧择保险网的，我们提供的服务不单单是售后问询，还包括即时的政策更新、保险知识普及等多方面内容，您添加企业微信后，可以第一时间获取最新的保险资讯和活动通知，确保您的保障不落后。那我这边用官方企业微信添加您，您那边同意一下就可以了，这样以后沟通起来更加方便哦。
用户:行吧
保险顾问:嗯嗯，太感谢您了，我已经发送了添加请求，您可以看一下微信通知，是不是有一个来自慧择保险的添加邀请。您点击同意，就可以加我为好友了，今后有关保险的问题或者需要服务，直接微信联系我，会更快捷哦。您看到了吗？
用户:没有
保险顾问:可能是网络延迟的原因，您可以稍后再看一下微信通知，或者进入微信的添加朋友页面，查看是否有来自慧择保险的请求。这样加上之后有任何问题都可以直接通过微信跟我沟通，我们会提供最及时的服务和回复给您。
用户:嗯
保险顾问:好的，您再检查一下您的微信聊天界面有没有收到一个黄色小喇叭的服务通知，通常添加好友的请求就是通过这个服务通知发出的。如果实在没有，可能是哪里出了小差错，您稍后再看一下。同时，如果您方便的话，我也可以再发送一次添加请求，确保您能收到。您觉得怎么样？这样之后我们通过微信交流会更加方便快捷哦。
用户:看到了
保险顾问:太好了，您点进去下面有一个企业微信的添加邀请，长按二维码添加到通讯录就可以了，很快的。您操作一下，正好我这边也为您做一下备注，以便后期能够更好地服务到您。您加上了可以跟我说一声。
用户:好，通过了
保险顾问:太感谢您的支持了！有了您的微信，我会及时为您解答任何保险方面的疑问，并分享一些实用的保险小知识。如果您以后遇到任何保险方面的问题或者需要咨询服务，记得找我哦，我会随时待命，为您提供专业的指导和帮助。再次感谢您的信任与支持！
"""
    hangup = HangUp(role_name='保险顾问',
                    conversation_history=conversation_history)
    hangup_tag = asyncio.run(hangup.achat_auto_llm(type="gpt"))
    print(hangup_tag)
